DTCA:扩散语言模型强化学习的去噪 Token 信用分配
🧭 扩散语言模型 RL 训练 · 信用分配
Zhejiang University · National University of Singapore · SAP
提出 DTCA,按去噪 token 置信度增益估计贡献并分组,用于扩散语言模型 RL 的 token 级信用分配。
📌 工作参考价值(4 点)
- 面向文案/query 生成的 RL 训练:若用 RL 优化广告文案、搜索词或商品标题生成,稀疏 outcome reward 会导致 credit assignment 困难。可借鉴 DTCA 的 token 贡献估计,用生成过程中 token 置信度增益作为细粒度奖励塑形,加速收敛。
- 按语言单元分组校准信号:DTCA 按 POS 分组贡献。在电商场景中,可对商品标题/query 中的品牌词、类目词、属性词分别设定权重,让 RL 更关注关键信息 token,减少语义噪声干扰。
- 熵变化校准避免过优化:策略更新引起的 token 熵变化被用于校准学习信号。实际 RL 训练生成模型时,可监控 token 级熵,抑制高熵 token 的过大梯度,提升训练稳定性。
- 筛选 informative 去噪 timesteps:DTCA 选择关键去噪步骤参与 RL 训练。若业务中采用扩散或非自回归生成模型,可减少无效训练步,节省算力并提升信号质量。
SPOT:通过提交顺序预测增强扩散语言模型的推测解码
🧭 扩散语言模型推理加速 · 推测解码
Technion – Israel Institute of Technology · Ben-Gurion University of the Negev
提出SPOT小模型预测扩散语言模型的下一个提交位置,构建候选树实现无损推测解码,吞吐最高提升3.4倍
📌 工作参考价值(4 点)
- 若业务使用扩散语言模型生成文本(如商品描述、推荐理由、query 扩展),可部署 SPOT 这类 0.62M 参数的 drafter,仅读目标模型输出分布预测更新顺序,目标模型单次 pass 验证,实现无损加速,额外计算成本极低。
- 非自回归生成场景(如并行生成多个推荐解释、多位置 mask 填充)可借鉴“顺序外置”思路:训练轻量模型预测哪些位置先提交,减少大模型前向次数,适合低延迟实时推荐。
- SPOT 的 drafter 只依赖目标模型输出分布,具备跨模型迁移能力,可以在多个不同扩散语言模型间复用,降低频繁换模型的部署成本;业务上可训练一个通用 drafter。
- SPOT-Sets 支持动态阈值解码下一次性提交多个位置,适合同时生成多个字段(标题、标签、摘要)的任务,在 SDAR-8B 上平均提升 59% 吞吐,可借鉴到多字段并行生成的工程实现。
不同任务不同去噪步:扩散语言模型的任务-时间感知量化
🧭 扩散语言模型 · 混合精度量化
Technion – Israel Institute of Technology · INSIGHT Lab, Ben-Gurion University of the Negev
发现扩散语言模型量化敏感性随任务与去噪步变化,提出无标签信号驱动的混合精度量化框架T2Quant
📌 工作参考价值(4 点)
- 若部署扩散语言模型做生成式推荐/Agent文本生成,可对去噪过程不同时间步做混合精度量化,而非全程统一位宽:在任务敏感步分配更高精度,其他步降低精度,能在固定显存/算力预算下提升生成质量。
- 采用无标签的敏感性信号(如全精度与量化模型输出分布的KL散度)离线为不同任务、不同时间步生成量化位宽策略,避免每个下游任务都需真实标注或评测指标,便于快速适配新业务场景。
- 该混合精度框架可与现有量化方法叠加(即插即用),工程上只需在推理时按步动态切换激活精度,不涉及模型重新训练;对需要反复去噪的扩散generation pipeline(如商品标题、推荐理由、Agent规划)可有效降低推理成本。
- 注意:这一结论源于扩散语言模型,若业务使用自回归LLM则借鉴有限;但若采用扩散或迭代解码架构,应避免沿用AR静态量化假设,需重新评估时间步/任务维度的敏感性。
扩散语言模型的多目标少样本提示
🧭 扩散语言模型 · 少样本多目标填充
University of Texas at Austin
用扩散语言模型在共享画布中同时填充多个掩码区域,并通过似然比奖励提升少样本示例跟随能力
📌 工作参考价值(4 点)
- 在电商文本结构化改写(如根据 2-3 个示例将商品描述转换成标题+卖点+属性标签等多字段)中,可借鉴扩散 LM 的多目标掩码填充:一次生成多个字段,利用左右上下文约束,比自回归逐 token 生成更连贯且避免重复。
- 选择性掩码+似然比奖励的思路可以迁移到 LLM 解码或 prompt 设计:只对关键槽位计算「有示例支持 vs 无示例支持」的概率比,作为过滤或重排分数,提高生成结果对 few-shot 示例的忠实度,减少随意发挥。
- 在需要格式转换/API 迁移的工程任务(如商品数据批量清洗、旧接口字段映射新接口)中,可以把每个字段当作一个掩码目标,展示少量成对示例后让扩散 LM 同时填充多个目标,减少多步串行生成的误差累积。
- 若团队已在探索生成式推荐中使用 diffusion LM 生成 Semantic ID 或多属性 token,这种共享画布的多目标填充可用来同时预测多个 token(如类目+属性+ID),相互提供上下文。
面向连续扩散语言模型的 Transformer 架构设计
🧭 连续扩散语言模型 · Transformer 架构优化
Polyadic Research Group · Dots Studio, Xiaohongshu · Peking University · SAGA Lab, The Chinese University of Hong Kong, Shenzhen · The University of California, Los Angeles
提出 LangDiT 架构,通过多项归一化与梯度路由改进,显著降低生成困惑度,刷新连续扩散语言模型 SOTA
📌 工作参考价值(4 点)
- 如果业务中尝试用扩散模型做文本生成(如商品描述、推荐理由、query 扩写),LangDiT 的 QK 归一化、更宽噪声条件和梯度路由技巧可直接迁移,降低生成困惑度。
- 连续扩散语言模型支持并行解码,适合低延迟文本生成场景,可探索用于实时推荐理由或广告文案生成。
- 置信度门控训练自条件可在训练时随机用模型自身预测作为条件,提升采样质量且不增加推理成本,适合作为生成式推荐模型的训练 trick。
- 架构级改进(而非单纯堆参数)对扩散生成质量提升明显,验证了在生成式模型中骨干网络设计的重要性,可借鉴到自研生成式推荐模型的 backbone 优化中。
SpecFold:折叠多分支冗余加速扩散语言模型推测解码
🧭 扩散语言模型 · 推测解码加速
Georgia Institute of Technology
利用多分支推测验证中的隐藏状态冗余,通过残差门控与折叠注意力/FFN降低计算,最高实现1.64倍吞吐提升
📌 工作参考价值(4 点)
- 若线上用 multi-branch speculative decoding 加速 LLM 生成(如 query 改写、商品文案、Agent 回复),可借鉴 SpecFold 的 parent-child 分支隐藏状态复用:draft branches 往往只额外 unmask 少量 token,大部分 hidden states 与父分支高度相似,可做 token-level residual gating 跳过重复 attention/FFN 计算。
- 算法-系统协同设计值得参考:将细粒度重用逻辑用 Triton kernel 实现为稀疏多分支执行,避免框架层 overhead,否则算法上的 FLOPs 节省难以转化为端到端吞吐增益。
- SpecFold 与 temporal caching 正交,可与 KV cache、前缀缓存等叠加;若未来在生成式推荐中使用扩散模型生成 Semantic ID 序列,同样存在跨步/跨分支冗余,可借鉴类似折叠思想。
- 论文验证了在多个 DLLM 上吞吐提升且精度不降,说明这种近似重用在文本生成任务上风险可控,适合延迟敏感的搜索/推荐在线服务。
面向高效推理:为扩散语言模型学习因果捷径
🧭 扩散语言模型推理训练优化
Zhejiang University · Shanghai AI Laboratory
提出CSL框架,通过提取因果捷径并优先掩码训练,提升扩散语言模型推理精度与收敛速度
📌 工作参考价值(4 点)
- 如果业务中尝试用 Diffusion LM 做推理型任务(如 Agent 规划、推荐理由生成),不要用 random masking;先提取推理链中的关键 token(causal shortcuts),优先 mask 这些 token 做 SFT,能明显提升收敛速度和最终精度。
- 论文给了一套 step-by-step token extraction 流程,可以从业务数据里自动挖出“因果捷径” token 链,适合迁移到需要生成结构化推理路径的场景(如电商导购 Agent 的决策链路)。
- 核心结论是:在双向扩散模型的 SFT 中,控制 mask 分布比随机 mask 更高效,这个思想也可以借鉴到其他 masked/denoising 序列模型训练中,作为训练效率调优手段。
- 对线上推理有启发:如果已部署 DLM,可以通过识别并保护 causal shortcut token 来减少探索空间,加快生成收敛,降低推理延迟。
扩散语言模型有界状态推理的寄存器令牌
🧭 扩散语言模型 · 有界状态推理 ★ 深度解读
University of Wisconsin–Madison · Microsoft Research · UC San Diego
用固定位置的寄存器 token 在分块生成间携带连续状态,优于离散文本携带,代码生成提升显著
📌 工作参考价值(3 点)
- 在长期会话、导购或任务型 Agent 中,若上下文预算受限,可借鉴 register token:在固定位置放少量可学习连续 slot,每轮结束后用一次额外 forward 把状态写入 slot,下一轮复用,避免无限拼接历史;训练时随机 mask prompt 并加入全掩码轮次,防止模型直接看原文绕过状态编码。
- 对分块生成长代码、营销文案或批量商品描述的场景,register carry 可替代 full-context,每 chunk 只多一次 forward,论文实验得到 3.4–5.6× wall-clock 加速,代码 pass@1 高于无 carry 的 full-seq SFT。
- RL 微调时,register 可作为可微状态通道:chunked GRPO 的 advantage 能跨 chunk 回传到 register 写入,适合长 horizon 奖励稀疏任务;同时低维 register 可用线性 probe 做状态审计,判断模型是否记住用户目标、任务进度等。
CanvasAnneal:扩散语言模型的课程式强化学习框架
🧭 扩散语言模型 · 课程强化学习
Google DeepMind
通过向扩散 RL 训练初期注入教师推理轨迹并逐步退火,缓解探索瓶颈,提升数学推理与工具调用性能
📌 工作参考价值(4 点)
- 在生成式推荐模型的 RL 训练中(如推荐理由、query 改写、对话式 Agent),早期探索难可用教师模型生成的高质量轨迹初始化 action/context,再逐步减少依赖,能显著加快收敛。
- 若业务采用 diffusion/非自回归生成(如并行生成候选序列、文案),可将教师轨迹作为初始 canvas 的 masked/部分可见条件,做课程退火,缓解探索瓶颈。
- 设计可退火的先验注入而非固定 prompt,监控 reward 提升曲线;该方法对多步推理、工具调用等困难任务更有效,简单任务收益有限,需评估任务难度。
- 工程上保存 teacher 轨迹作为额外输入,注意训练/推理解耦:推理时不再提供教师轨迹,避免分布不一致。
潜隐时钟:扩散语言模型中的隐式时间建模
🧭 扩散语言模型的可解释性分析
Sapienza University of Rome · EPFL · NVIDIA
发现无显式时间条件的扩散语言模型内部编码了隐式去噪时间步,可通过探针提取和方向操控来调制模型行为
📌 工作参考价值(4 点)
- 对于使用扩散模型生成推荐内容(如GenRec)的场景,可借鉴隐式时间步操控来动态调节生成的确定性与多样性:沿时间步子空间方向移动可改变模型置信度,实现从保守推荐到探索式推荐的切换。
- 工程实现上,可在推理时通过简单的激活向量加法(无须重新训练)来控制生成进度,用于A/B测试或个性化生成风格。
- 探针方法可用于监控线上模型的去噪状态,例如检测异常生成或提前终止去噪步骤以节省计算。
- 若将扩散步骤与推荐质量关联,可利用隐式时间步表征设计自适应调度策略,在低延迟场景下跳过部分去噪步骤。
扩散语言模型的自适应多步前瞻解码
🧭 扩散语言模型自适应解码优化
Michigan State University · Morgan Stanley · IBM T.J. Watson Research Center
提出自适应前瞻解码框架,动态决定前瞻深度,在扩散语言模型上实现更优的准确率-步数权衡
📌 工作参考价值(4 点)
- 自适应前瞻机制可迁移至多步Agent决策:当中间状态不确定性高时触发更深入的规划,避免固定深度造成的浪费或不足。
- 基于候选评分方差的动态探索停止策略,可直接用于推荐系统的多步重排序或生成式推荐的解码过程,节省计算量。
- 并行解码与前瞻结合的思路,可启发电商搜索中的批量query生成或文案生成,在效率与质量间取得更好平衡。
- 从中间状态重新触发前瞻的机制,类似于Agent在任务执行中根据中间结果重新规划,可用于复杂推荐流程的实时调整。
掩码扩散语言模型的双向归纳电路与隐式时间步分析
🧭 扩散语言模型机制分析 · 上下文学习
National University of Science and Technology POLITEHNICA Bucharest
发现扩散语言模型用双向归纳电路实现上下文学习,且以全局掩码比例作隐式时间步
📌 工作参考价值(4 点)
- 对于电商搜索中的序列建模,可借鉴双向上下文访问的思路:在允许未来信息的场景(如离线重排)下,利用扩散模型隐式地融合下文,提升对用户意图的推断准确性。
- 隐式时间步的发现提示:在设计扩散式推荐或生成式检索时,可以省略显式时间步嵌入,让模型从噪声比例中自学习去噪进度,简化工程实现。
- 归纳电路的分析方法(激活修补、因果追踪)可以迁移到推荐模型的解释性工作中,例如定位关键注意力头,帮助 debug 召回结果中的主观偏差。
- 当仅左上下文可用时,DLM 的 induction 能力并不优于自回归模型,因此若线上真实场景只能获取左侧历史行为,盲目切换生成范式未必带来增益,需根据上下文可获取性选择架构。
多块扩散语言模型
🧭 扩散语言模型 · 多块并行解码
Shanghai Jiao Tong University · Xi'an Jiao Tong University · Huawei
通过多块教师强制训练和块缓冲解码,将块扩散语言模型扩展至多块并行生成,实现近两倍加速且精度不降
📌 工作参考价值(4 点)
- 若在推荐系统中用扩散模型生成文案、推荐理由等长文本,可借鉴多块并行解码思路,将生成过程拆分为固定大小的噪声块组,提升并行度,降低推理延迟。
- 块缓冲(Block Buffer)机制通过保持 KV cache 静态形状和前缀复用,能直接迁移到在线 LLM 服务的缓存管理,减少显存波动,适合高并发场景。
- 多块教师强制(MultiTF)提供了一种低成本后训练方案:在已有 BD-LM 基础上,只需构造带噪声的连续块与干净前缀的混合训练数据,就可以让模型适配更高效的解码策略,适合快速迭代。
- 实验中的 TPF 指标可作为工程权衡的参考:在精度允许范围内(如 1% 准确率损失),可将每步生成令牌数提升近 3 倍,有利于在成本与效果间做配置化调优。
扩散语言模型的自增强检索
🧭 扩散语言模型 · 动态检索增强
Cornell University
利用扩散去噪中丢弃的低置信度 token 作为前瞻信号,实现无训练的动态检索增强生成
📌 工作参考价值(4 点)
- 电商对话或商品描述生成中,可利用扩散模型的并行去噪过程,提前从中间低置信度 token 中提取关键实体,触发多跳知识检索,提升答案的完整性和事实准确性。
- 方法完全无训练、不依赖特定检索器,可直接嵌入现有的扩散语言模型推理流程,非常适合业务中快速尝试不同的知识库或检索策略。
- 扩散模型的高吞吐特性(8× 提升)加上前瞻检索,能在高并发电商客服或内容生成场景下,在保持答案质量的同时显著降低延迟。
- 对于 Agent 中的多跳推理或工具调用,可以借鉴前瞻 token 的思路,在最终决策前动态获取更多环境信息,提升决策可靠性。